#include "nrf24l01.h"
#include "spi.h"

const uint8_t TX_ADDRESS[TX_ADR_WIDTH]={0x34,0x43,0x10,0x10,0x01}; //发送地址
const uint8_t RX_ADDRESS[RX_ADR_WIDTH]={0x34,0x43,0x10,0x10,0x01}; //发送地址

void nrf_ce(uint8_t b)
{
    if(b) {
        HAL_GPIO_WritePin(RF_CE_GPIO_Port,RF_CE_Pin,GPIO_PIN_SET);
    }
    else {
        HAL_GPIO_WritePin(RF_CE_GPIO_Port,RF_CE_Pin,GPIO_PIN_RESET);
    }
}

void nrf_csn(uint8_t b)
{
    if(b)  {
        HAL_GPIO_WritePin(RF_CSN_GPIO_Port,RF_CSN_Pin,GPIO_PIN_SET);
    }
    else  {
        HAL_GPIO_WritePin(RF_CSN_GPIO_Port,RF_CSN_Pin,GPIO_PIN_RESET);
    }
}

uint8_t nrf_read_irq(void)
{
    return HAL_GPIO_ReadPin(RF_IRQ_GPIO_Port,RF_IRQ_Pin);
}
    
uint8_t nrf_spi_read_write(uint8_t data)
{
    uint8_t rx_data;
    HAL_SPI_TransmitReceive(&hspi1,&data,&rx_data,1,1000);
    return rx_data;
}


//SPI写寄存器
//reg:指定寄存器地址
//value:写入的值
uint8_t NRF24L01_Write_Reg(uint8_t reg,uint8_t value)
{
	uint8_t status;	
    nrf_csn(0);                 //使能SPI传输
  	status =nrf_spi_read_write(reg);//发送寄存器号 
  	nrf_spi_read_write(value);      //写入寄存器的值
  	nrf_csn(1);                 //禁止SPI传输	   
  	return(status);       			//返回状态值
}

//读取SPI寄存器值
//reg:要读的寄存器
uint8_t NRF24L01_Read_Reg(uint8_t reg)
{
	uint8_t reg_val;	    
 	nrf_csn(0);               //使能SPI传输		
  	nrf_spi_read_write(reg);   //发送寄存器号
  	reg_val=nrf_spi_read_write(0XFF);//读取寄存器内容
  	nrf_csn(1);          //禁止SPI传输		    
  	return(reg_val);           //返回状态值
}	

//在指定位置读出指定长度的数据
//reg:寄存器(位置)
//*pBuf:数据指针
//len:数据长度
//返回值,此次读到的状态寄存器值 
uint8_t NRF24L01_Read_Buf(uint8_t reg,uint8_t *pBuf,uint8_t len)
{
	uint8_t status,u8_ctr;	       
  	nrf_csn(0);           //使能SPI传输
  	status=nrf_spi_read_write(reg);//发送寄存器值(位置),并读取状态值   	   
 	for(u8_ctr=0;u8_ctr<len;u8_ctr++)
    {
        pBuf[u8_ctr]=nrf_spi_read_write(0XFF);//读出数据
    }
  	nrf_csn(1);       //关闭SPI传输
  	return status;        //返回读到的状态值
}

//在指定位置写指定长度的数据
//reg:寄存器(位置)
//*pBuf:数据指针
//len:数据长度
//返回值,此次读到的状态寄存器值
uint8_t NRF24L01_Write_Buf(uint8_t reg, uint8_t *pBuf, uint8_t len)
{
	uint8_t status,u8_ctr;	    
 	nrf_csn(0);          //使能SPI传输
  	status = nrf_spi_read_write(reg);//发送寄存器值(位置),并读取状态值
  	for(u8_ctr=0; u8_ctr<len; u8_ctr++)
    {
        nrf_spi_read_write(*pBuf++); //写入数据	 
    }
  	nrf_csn(1);       //关闭SPI传输
  	return status;          //返回读到的状态值
}		

/*
    初始化NRF24L01
*/
void NRF24L01_Init(void)
{ 	
    nrf_ce(0);
    nrf_csn(1);	 	

  	NRF24L01_Write_Buf(WRITE_REG_NRF+TX_ADDR,(uint8_t*)TX_ADDRESS,TX_ADR_WIDTH);//写TX节点地址 
  	NRF24L01_Write_Buf(WRITE_REG_NRF+RX_ADDR_P0,(uint8_t*)RX_ADDRESS,RX_ADR_WIDTH);//写RX节点地址  
  	NRF24L01_Write_Reg(WRITE_REG_NRF+EN_AA,0x00);     //禁止自动应答    
  	NRF24L01_Write_Reg(WRITE_REG_NRF+EN_RXADDR,0x01); //使能通道0的接收地址  
  	NRF24L01_Write_Reg(WRITE_REG_NRF+RX_PW_P0,RX_PLOAD_WIDTH);//选择通道0的有效数据宽度     
  	NRF24L01_Write_Reg(WRITE_REG_NRF+SETUP_RETR,0x00);//禁止自动重发
  	NRF24L01_Write_Reg(WRITE_REG_NRF+RF_SETUP,0x0f);  //设置TX发射参数,0db增益,2Mbps,低噪声增益开启   	  		      
    
  	NRF24L01_Write_Reg(WRITE_REG_NRF+CONFIG, 0x0f);//配置基本工作模式的参数;PWR_UP,EN_CRC,16BIT_CRC,接收模式 

}

//检测24L01是否存在
//返回值:0，成功;1，失败	
uint8_t NRF24L01_Check(void)
{
	uint8_t buf[5]={0XA5,0XA5,0XA5,0XA5,0XA5};
	uint8_t i;
  	 
	NRF24L01_Write_Buf(WRITE_REG_NRF+TX_ADDR,buf,5);//写入5个字节的地址.	
    
    HAL_Delay(10);
    
	NRF24L01_Read_Buf(TX_ADDR,buf,5); //读出写入的地址  
	for(i=0;i<5;i++)
    {
        if(buf[i]!=0XA5)break;	 							   
    }
    
	if(i!=5)return 1;//检测24L01错误	
	return 0;		 //检测到24L01
}	 	 
		   
//启动NRF24L01发送一次数据
//txbuf:待发送数据首地址
//返回值:发送完成状况
uint8_t NRF24L01_TxPacket(uint8_t *txbuf)
{
	uint8_t sta;
    
	nrf_ce(0);
  	NRF24L01_Write_Buf(WR_TX_PLOAD,txbuf,TX_PLOAD_WIDTH);//写数据到TX BUF  32个字节
 	nrf_ce(1);//启动发送	   
	while(nrf_read_irq()!=0);//等待发送完成
    
	sta=NRF24L01_Read_Reg(STATUS);  //读取状态寄存器的值	   
	NRF24L01_Write_Reg(WRITE_REG_NRF+STATUS,sta); //清除TX_DS或MAX_RT中断标志
    
	if(sta&TX_OK)//发送完成
	{
		return TX_OK;
	}
    
	return 0xff;//其他原因发送失败
}

//启动NRF24L01接收一次数据
//txbuf:待发送数据首地址
//返回值:0，接收完成；其他，错误代码
uint8_t NRF24L01_RxPacket(uint8_t *rxbuf)
{
	uint8_t sta;		    							   
    
	sta=NRF24L01_Read_Reg(STATUS);  //读取状态寄存器的值    	 
	NRF24L01_Write_Reg(WRITE_REG_NRF+STATUS,sta); //清除TX_DS或MAX_RT中断标志
    
	if(sta&RX_OK)//接收到数据
	{
		NRF24L01_Read_Buf(RD_RX_PLOAD,rxbuf,RX_PLOAD_WIDTH);//读取数据
		NRF24L01_Write_Reg(FLUSH_RX,0xff);//清除RX FIFO寄存器 
		return 0; 
	}	   
    
	return 1;//没收到任何数据
}					    

/*
    设置RF通信频率
*/
void NRF24L01_SetChannel(uint8_t ch)
{
  	NRF24L01_Write_Reg(WRITE_REG_NRF+RF_CH,ch);	     	
}

//该函数初始化NRF24L01到RX模式
//设置RX地址,写RX数据宽度,选择RF频道,波特率和LNA HCURR
//当CE变高后,即进入RX模式,并可以接收数据了		   
void NRF24L01_RX_Mode(uint8_t channel)
{
	nrf_ce(0);	  
  	NRF24L01_Write_Buf(WRITE_REG_NRF+RX_ADDR_P0,(uint8_t*)RX_ADDRESS,RX_ADR_WIDTH);//写RX节点地址
	  
  	//NRF24L01_Write_Reg(WRITE_REG_NRF+EN_AA,0x00);    //禁止自动应答    
  	NRF24L01_Write_Reg(WRITE_REG_NRF+EN_RXADDR,0x01);//使能通道0的接收地址  		  
  	//NRF24L01_Write_Reg(WRITE_REG_NRF+RX_PW_P0,RX_PLOAD_WIDTH);//选择通道0的有效数据宽度 
    NRF24L01_SetChannel(channel);
  	//NRF24L01_Write_Reg(WRITE_REG_NRF+RF_SETUP,0x0f);//设置TX发射参数,0db增益,2Mbps,低噪声增益开启   
  	NRF24L01_Write_Reg(WRITE_REG_NRF+CONFIG, 0x0f);//配置基本工作模式的参数;PWR_UP,EN_CRC,16BIT_CRC,接收模式 
  	nrf_ce(1); //CE为高,进入接收模式 
}		

//该函数初始化NRF24L01到TX模式
//设置TX地址,写TX数据宽度,设置RX自动应答的地址,填充TX发送数据,选择RF频道,波特率和LNA HCURR
//PWR_UP,CRC使能
//当CE变高后,即进入RX模式,并可以接收数据了		   
//CE为高大于10us,则启动发送.	 
void NRF24L01_TX_Mode(uint8_t channel)
{														 
	nrf_ce(0);	    
  	NRF24L01_Write_Buf(WRITE_REG_NRF+TX_ADDR,(uint8_t*)TX_ADDRESS,TX_ADR_WIDTH);//写TX节点地址 
  	//NRF24L01_Write_Buf(WRITE_REG_NRF+RX_ADDR_P0,(uint8_t*)RX_ADDRESS,RX_ADR_WIDTH); //设置TX节点地址,主要为了使能ACK	  

  	//NRF24L01_Write_Reg(WRITE_REG_NRF+EN_AA,0x00);     //禁止自动应答    
  	//NRF24L01_Write_Reg(WRITE_REG_NRF+EN_RXADDR,0x01); //使能通道0的接收地址  
  	//NRF24L01_Write_Reg(WRITE_REG_NRF+SETUP_RETR,0x00);//禁止自动重发
  	NRF24L01_SetChannel(channel);
  	//NRF24L01_Write_Reg(WRITE_REG_NRF+RF_SETUP,0x0f);  //设置TX发射参数,0db增益,2Mbps,低噪声增益开启   
  	NRF24L01_Write_Reg(WRITE_REG_NRF+CONFIG,0x0e);    //配置基本工作模式的参数;PWR_UP,EN_CRC,16BIT_CRC,接收模式,开启所有中断
	nrf_ce(1);//CE为高,10us后启动发送
}		  




